"""
Author: yida
Time is: 2022/4/26 20:58 
this Code: 统计模型参数
"""
from torchstat import stat
from torchvision.models import resnet18
from torchsummary import summary
import torchstat

if __name__ == '__main__':
    model = resnet18(pretrained=False, num_classes=9)
    stat(model, (3, 600, 600))

    # 导入模型，输入一张输入图片的尺寸
    summary(model, input_size=(3, 600, 600), batch_size=64)
